jax>=0.2.0
tensorflow>=1.11.0
numpy>=1.13.1
absl-py>=0.1.6
